package cn.lianyutiantang.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**压缩文件工具类
 * Created by zhengjingli on 2017/4/6.
 */
public class ZipUtils {
    public static void doCompress(String srcFile, String zipFile) throws Exception {
        doCompress(new File(srcFile), new File(zipFile));
    }

    /**
     * 文件压缩
     * @param srcFile  目录或者单个文件
     * @param destFile 压缩后的ZIP文件
     */
    public static void doCompress(File srcFile, File destFile) throws Exception {
        ZipOutputStream out = new ZipOutputStream(new FileOutputStream(destFile));
        if(srcFile.isDirectory()){
            File[] files = srcFile.listFiles();
            for(File file : files){
                doCompress(file.getName(),file, out);
            }
        }else {
            doCompress("默认名称",srcFile, out);
        }
    }

    public static void doCompress(String fileName,String pathname, ZipOutputStream out) throws IOException{
        doCompress(fileName,new File(pathname), out);
    }

    public static void doCompress(String fileName,File file, ZipOutputStream out) throws IOException{
        if( file.exists() ){
            byte[] buffer = new byte[1024];
            FileInputStream fis = new FileInputStream(file);
            //out.putNextEntry(new ZipEntry(URLEncoder.encode(file.getName(), "UTF-8")/*file.getName()*/));
            //out.putNextEntry(new ZipEntry(file.getName()));
            out.putNextEntry(new ZipEntry(fileName));
            int len = 0 ;
            // 读取文件的内容,打包到zip文件
            while ((len = fis.read(buffer)) > 0) {
                out.write(buffer, 0, len);
            }
            out.flush();
            out.closeEntry();
            fis.close();
        }
    }
}
